python藏头诗生成器

春节前用 GPT2 训练了一个自动对联系统：鼠年春节，用 GPT-2 自动生成(写)春联和对对联 ，逻辑上来说这套NLG方法论可以应用于任何领域文本的自动生成，当然，格式越固定越好，这让我自然想到了自动写诗词，诗词的格式相对比较固定，我们之前已经有所涉及，譬如已经在AINLP公众号上上线了自动写藏头诗和首字诗的功能，不过是直接复用的："自动作诗机"上线，代码和数据都是公开的 ，另外还有一个更大的诗词数据项目可以用作自动作诗的“原料”：【Github】Chinese-poetry: 最全中华古诗词数据库，加上 GPT2-Chinese 这个项目：【Github】GPT2-Chinese：中文的GPT2训练代码 ，可以说万事俱备，只欠试用。

所以本周我们从五言绝句开始继续自然语言生成的主题，关于五言绝句，百度百科是这样说的：

五言绝句是中国传统诗歌的一种体裁，简称五绝，是指五言四句而又合乎律诗规范的小诗，属于近体诗范畴。此体源于汉代乐府小诗，深受六朝民歌影响，成熟定型于唐代。五绝每首仅二十字，便能展现出一幅幅清新的图画，传达一种种真切的意境。因小见大，以少总多，在短章中包含着丰富的内容，是其最大特色。五绝有仄起、平起二格。代表作品有王维的《鸟鸣涧》、李白的《静夜思》、杜甫的《八阵图》、王之涣的《登鹳雀楼》、刘长卿的《送灵澈上人》等。

我主要用了 Chinese-poetry 里的《全唐诗》和《全宋诗》数据 ，首先向这个项目的作者致敬：

《全唐诗》是清康熙四十